(function() {
    'use strict';
    
    // 获取显示元素
    const timeElement = document.querySelector('.right-mid-content .right-top-content:nth-child(1) .text');
    const speedElement = document.querySelector('.right-mid-content .right-top-content:nth-child(2) .text');
    const flowElement = document.querySelector('.right-mid-content .right-top-content:nth-child(3) .text');
    const congestionElement = document.querySelector('.right-mid-content .right-top-content:nth-child(4) .text');
    
    // 拥堵等级对应的颜色
    const congestionColors = {
        1: '#336838',  // 畅通
        2: '#8BBD63',  // 基本畅通
        3: '#FEE88F',  // 轻度拥堵
        4: '#FCCA60',  // 中度拥堵
        5: '#E04B48',  // 严重拥堵
        6: '#A61A2C'   // 特别严重拥堵
    };
    
    // 更新数据的函数
    function updatePredictionData() {
        fetch('http://localhost:5000/prediction')
            .then(response => response.json())
            .then(data => {
                // 更新时间
                timeElement.textContent = data.time;
                
                // 更新预测车速
                speedElement.textContent = data.speed + ' km/h';
                
                // 更新预测流量
                flowElement.textContent = data.flow + ' 辆/小时';
                
                // 更新拥堵等级
                congestionElement.textContent = data.congestion;
                congestionElement.style.color = congestionColors[data.congestion] || '#fff';
            })
            .catch(error => console.error('获取预测数据失败:', error));
    }
    
    // 首次加载
    updatePredictionData();
    
    // 每10秒更新一次数据
    setInterval(updatePredictionData, 10000);
})(); 